from ultralytics import YOLO

model=YOLO("/home/tuchunxu/workspace/cv_project/runs/pose/train/weights/best.pt")

res=model.predict("/home/tuchunxu/workspace/cv_project/datasets/cat_pose/CAT_06/00001441_010.jpg")

res[0].save("test.jpg")
# print(res)